// 四皇后问题 回溯法经典

#include <stdio.h>
#include <stdlib.h>
#define max 4
// 递归解决
int map[max][max] = {0};
// 检查i,j是否合法
int check(int i, int j)
{
    // i,j所在位置 同一行 同一列 所在一对角线不得有棋子
    for (int m = 0; m < max; m++)
    {
        for (int n = 0; n < max; n++)
        {
            // 同一行 一列
            if ((m == i && map[m][n] == 1) || (n == j && map[m][n] == 1))
                return 0;
            // 同一对角线
            if ((m + n == i + j && map[m][n] == 1) || (m - n == i - j && map[m][n] == 1))
                return 0;
        }
    }
    return 1;
}
int cont = 0;
void queen(int m)
{
    if (m == max)
    {
        printf("%dth time:\n", ++cont);
        // 输出棋盘
        for (int i = 0; i < max; i++)
        {
            for (int j = 0; j < max; j++)
            {
                printf("%3d", map[i][j]);
            }
            printf("\n");
        }
        printf("\n");
    }
    for (int k = 0; k < max; k++)
    {
        if (check(m, k))
        {
            // 放置一颗棋子
            map[m][k] = 1;
            queen(m + 1);
            // 撤销这颗棋子
            map[m][k] = 0;
        }
    }
}

int main()
{
    /*******************code*******************/
    queen(0);
    /******************************************/
    printf("\n\n****************************\n");
    printf("Press Enter key to continue\n");
    getchar();
    return 0;
    /******************************************/
}
